'use strict';

// 自定义 401 中间件
module.exports = () => {
  return async function(ctx, next) {
    await next();
    if (ctx.status === 401) {
      ctx.set('Content-Type', 'text/html');
      await ctx.render('401');
    }
  };
};
